import java.util.Comparator;
import java.util.PriorityQueue;

//Top-k问题
class Solution {
    public int[] smallestK(int[] arr, int k){
        //判断k是否大于1，数组是否为空
        if(arr.length==0||k<1){
            return  new int[0];
        }
        PriorityQueue<Integer> p1=new PriorityQueue<>
                (k, new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o2.compareTo(o1);
            }
        });
        //将前k个元素，建成大堆
        for (int i = 0; i <k ; i++) {
            p1.offer(arr[i]);
        }
        //把剩下的元素和堆顶元素相比
        for (int i = k; i <arr.length ; i++) {
            int tmp = p1.peek();
            if (tmp < arr[i]) {
                p1.poll();
                p1.offer(arr[i]);
            }
        }
            //将堆的元素赋值到数组里
        int [] ret=new int[k];
        for (int i = 0; i <k ; i++) {
            ret[i]= p1.poll();
        }
        return ret;
    }
  

}